#include <linux/head.h>
#include <linux/sched.h>
#include <asm/system.h>
#include <asm/io.h>

/*
 // TODO ： 部分实现涉及 sys_calls.s
 // TODO：后续实现
 */

static void die(char * str, long esp_ptr, long nr)
{
	/* TODO:待后续实现*/
	// long * esp = (long *) esp_ptr;
	// int i;

	// printk("%s: %04x\n\r",str,nr&0xffff);
	// printk("EIP:\t%04x:%p\nEFLAGS:\t%p\nESP:\t%04x:%p\n",
	// 	esp[1],esp[0],esp[2],esp[4],esp[3]);
	// printk("fs: %04x\n",_fs());
	// printk("base: %p, limit: %p\n",get_base(current->ldt[1]),get_limit(0x17));
	// if (esp[4] == 0x17) {
	// 	printk("Stack: ");
	// 	for (i=0;i<4;i++)
	// 		printk("%p ",get_seg_long(0x17,i+(long *)esp[3]));
	// 	printk("\n");
	// }
	// str(i);
	// printk("Pid: %d, process nr: %d\n\r",current->pid,0xffff & i);
	// for(i=0;i<10;i++)
	// 	printk("%02x ",0xff & get_seg_byte(esp[1],(i+(char *)esp[0])));
	// printk("\n\r");
	// do_exit(11);		/* play segment exception */
}

void _divide_error(void);
#define divide_error _divide_error
void _debug(void);
#define debug _debug
void _nmi(void);
#define nmi _nmi
void _int3(void);
#define int3 _int3
void _overflow(void);
#define overflow _overflow
void _bounds(void);
#define bounds _bounds
void _invalid_op(void);
#define invalid_op _invalid_op
void _double_fault(void);
#define double_fault _double_fault
void _coprocessor_segment_overrun(void);
#define coprocessor_segment_overrun _coprocessor_segment_overrun
void _invalid_TSS(void);
#define invalid_TSS _invalid_TSS
void _segment_not_present(void);
#define segment_not_present _segment_not_present
void _stack_segment(void);
#define stack_segment _stack_segment
void _general_protection(void);
#define general_protection _general_protection
void coprocessor_error(void);
void device_not_available(void);
// void _irq13(void);
// #define irq13 _irq13
void _reserved(void);
#define reserved _reserved
void _alignment_check(void);
#define alignment_check _alignment_check

void do_divide_error(long esp, long error_code)
{
	die("divide error", esp, error_code);
}

void do_int3(long * esp, long error_code,
	long fs,long es,long ds,
	long ebp,long esi,long edi,
	long edx,long ecx,long ebx,long eax)
{
	// int tr;

	// __asm__("str %%ax":"=a" (tr):"0" (0));
	// printk("eax\t\tebx\t\tecx\t\tedx\n\r%8x\t%8x\t%8x\t%8x\n\r",
	// 	eax,ebx,ecx,edx);
	// printk("esi\t\tedi\t\tebp\t\tesp\n\r%8x\t%8x\t%8x\t%8x\n\r",
	// 	esi,edi,ebp,(long) esp);
	// printk("\n\rds\tes\tfs\ttr\n\r%4x\t%4x\t%4x\t%4x\n\r",
	// 	ds,es,fs,tr);
	// printk("EIP: %8x   CS: %4x  EFLAGS: %8x\n\r",esp[0],esp[1],esp[2]);
}

void do_nmi(long esp, long error_code)
{
	die("nmi",esp,error_code);
}

void do_overflow(long esp, long error_code)
{
	die("overflow",esp,error_code);
}

void do_bounds(long esp, long error_code)
{
	die("bounds",esp,error_code);
}

void do_invalid_op(long esp, long error_code)
{
	die("invalid operand",esp,error_code);
}

void do_device_not_available(long esp, long error_code)
{
	die("device not available",esp,error_code);
}

void do_coprocessor_segment_overrun(long esp, long error_code)
{
	die("coprocessor segment overrun",esp,error_code);
}

void do_invalid_TSS(long esp,long error_code)
{
	die("invalid TSS",esp,error_code);
}

void do_segment_not_present(long esp,long error_code)
{
	die("segment not present",esp,error_code);
}

void do_stack_segment(long esp,long error_code)
{
	die("stack segment",esp,error_code);
}

void do_double_fault(long esp, long error_code)
{
	die("double fault",esp,error_code);
}

void do_general_protection(long esp, long error_code)
{
	die("general protection",esp,error_code);
}

void do_alignment_check(long esp, long error_code)
{
    die("alignment check",esp,error_code);
}

void do_coprocessor_error(long esp, long error_code)
{
	/* TODO */
	// if (last_task_used_math != current)
	// 	return;
	// die("coprocessor error",esp,error_code);
}

void do_reserved(long esp, long error_code)
{
	die("reserved (15,17-47) error",esp,error_code);
}

void trap_init(void)
{
	int i;
	/* 0号中断是 除0错误，在 kernel/asm.s 中实现 */
	/*
	 * 0号中断是 除0错误，处理器使用
	 * DIV或IDIV指令的除数操作数是0；CS:EIP指向异常的指令
	 * 无错误码
	 */
	set_trap_gate(0,&divide_error);
	/* 1号中断是 debug 调试异常，无异常码 */
	set_trap_gate(1,&debug);
	/* 2号中断是 NMI中断： 不可屏蔽外部中断，无异常码*/
	set_trap_gate(2,&nmi);

	/* 3-5 号中断在用户态也可以调用，分别是 断点 溢出 和越界；无错误码 */
	set_system_gate(3,&int3);	/* int3-5 can be called from all */
	set_system_gate(4,&overflow);
	set_system_gate(5,&bounds);

	/* 6号中断是不合法的操作数，非法操作码（未定义操作码）；无错误码 */
	set_trap_gate(6,&invalid_op);
	/* 7号中断是设备不可用（无数学协处理器）；无错误码 */
	set_trap_gate(7,&device_not_available);
	/* 8号中断是双故障；有错误码 */
	set_trap_gate(8,&double_fault);
	/* 9号中断是协处理器段超出；无错误码 */
	set_trap_gate(9,&coprocessor_segment_overrun);
	/* 10号中断是非法TSS；有错误码 */
	set_trap_gate(10,&invalid_TSS);
	/* 11号中断是段不存在；有错误码 */
	set_trap_gate(11,&segment_not_present);
	/* 12号中断是栈段故障；有错误码 */
	set_trap_gate(12,&stack_segment);
	/* 13号中断是通用保护；有错误码 */
	set_trap_gate(13,&general_protection);
	// /* 14号中断是页故障错误：这个很重要，是实现写时复制的基础 */
	// set_trap_gate(14,&page_fault);
	/* 15号中断是保留 */
	set_trap_gate(15,&reserved);
	/* 16号中断是协处理器错误，x87 FPU符点错误（数学故障） */
	set_trap_gate(16,&coprocessor_error);
	/* 17号中断是对齐检查错误 */
	set_trap_gate(17,&alignment_check);
	// /* 18-47号中断也保留 */
	for (i=18;i<48;i++)
		set_trap_gate(i,&reserved);
	// /* 45号中断是 irq13 */
	// set_trap_gate(45,&irq13);

	/* 
	 * 这个是在处理8259A中断处理器控制 
	 * OCW1操作
	 * 	0xfb = 0b11111011	// 屏蔽主8259A的IRQ2 --- 屏蔽从芯片中断
	 * 	0xdf = 0b11011111	// 屏蔽从8259A的IRQ13 --- 数学协处理器中断
	 * 
	 */
	outb_p(inb_p(0x21)&0xfb,0x21);
	outb(inb_p(0xA1)&0xdf,0xA1);

	// /* 39号中断是并行中断 */
	// set_trap_gate(39,&parallel_interrupt);
}